﻿// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
// Copyright (c) 2011, Dr. Masroor Ehsan
// All rights reserved.
// 
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions are met:
//     * Redistributions of source code must retain the above copyright
//       notice, this list of conditions and the following disclaimer.
//     * Redistributions in binary form must reproduce the above copyright
//       notice, this list of conditions and the following disclaimer in the
//       documentation and/or other materials provided with the distribution.
//     * Neither the name of the author nor the
//       names of its contributors may be used to endorse or promote products
//       derived from this software without specific prior written permission.
// 
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
// ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
// WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
// DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
// DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
// (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
// LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
// ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
// SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
// 
//  $Author:$
//  $Id:$
//  $Rev:$
// 
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *

using System;
using System.Collections.Generic;
using System.Text;

namespace MaxTrader.IntradayQuotes.Core
{
    /// <summary>
    ///   CompositeIntradayStockSnapshots is a container of IntradayStockSnapshots
    /// </summary>
    public class CompositeIntradayStockSnapshots
    {
        public StockExchange Exchange { get; private set; }

        private readonly SortedList<string, IntradayStockSnapshots> _dictSnapshots;

        public CompositeIntradayStockSnapshots(StockExchange exchange)
        {
            Exchange = exchange;
            _dictSnapshots = new SortedList<string, IntradayStockSnapshots>();
        }

        public AggregateStockSnapshot LastQuotesForAllStocks()
        {
            var list = new AggregateStockSnapshot();
            foreach (var kvp in _dictSnapshots)
            {
                if (kvp.Value.LastQuote() != null)
                    list.Add(kvp.Key, new IntradayQuoteDetailed(kvp.Value));
            }
            return list;
        }

        public IntradayQuote LastQuote(string symbol)
        {
            return GetSnapshot(symbol).LastQuote();
        }

        private IntradayStockSnapshots GetSnapshot(string symbol, bool createNew = true)
        {
            symbol = symbol.Trim().ToUpperInvariant();
            if (!_dictSnapshots.ContainsKey(symbol))
            {
                if (!createNew) return null;
                _dictSnapshots.Add(symbol, new IntradayStockSnapshots(symbol, Exchange));
            }

            return _dictSnapshots[symbol];
        }

#if TRACK_TICK_SEQUENCE
        public void Add(string symbol, DateTime utcTime, decimal lastPrice, int cumVol, int volDiff = 0, int seqNum = 0)
#else
        public void Add(string symbol, DateTime utcTime, decimal lastPrice, int cumVol, int volDiff = 0)
#endif
        {
            var snapshot = GetSnapshot(symbol);
#if TRACK_TICK_SEQUENCE            
            snapshot.Add(utcTime, lastPrice, cumVol, volDiff, seqNum);
#else
            snapshot.Add(utcTime, lastPrice, cumVol, volDiff);
#endif
        }

#if TRACK_TICK_SEQUENCE
        public void Add(string symbol, DateTime utcTime, decimal lastPrice, decimal high, decimal low, int cumVol, int volDiff = 0, int seqNum = 0)
#else
        public void Add(string symbol, DateTime utcTime, decimal lastPrice, decimal high, decimal low, int cumVol,
                        int volDiff = 0)
#endif
        {
            var snapshot = GetSnapshot(symbol);
#if TRACK_TICK_SEQUENCE            
            snapshot.Add(utcTime, lastPrice, cumVol, volDiff, seqNum);
#else
            snapshot.Add(utcTime, lastPrice, cumVol, volDiff);
#endif
            snapshot.High = high;
            snapshot.Low = low;
        }

        public int SymbolsCount
        {
            get { return _dictSnapshots.Count; }
        }

        public IEnumerable<string> Symbols()
        {
            return _dictSnapshots.Keys;
        }

        public IntradayStockSnapshots Snapshot(string symbol)
        {
            return GetSnapshot(symbol, false);
        }

        public string DumpAll()
        {
            var sb = new StringBuilder();
            foreach (var snap in _dictSnapshots.Values)
            {
                foreach (var q in snap.GetQuotes())
                {
                    sb.AppendLine(q.ToString());
                }
            }
            return sb.ToString();
        }
    }
}